import matplotlib
import matplotlib.pyplot as plt
import numpy as np
matplotlib.use('TkAgg')
x = np.linspace(-3, 3, 50)
y1 = 2 * x + 1
y2 = x ** 2

plt.figure()

plt.plot(x, y2)
plt.plot(x, y1, color='red', linewidth=1.0, linestyle='--')
#
# # 使用plt.xlim设置x坐标轴范围：(-1, 2)； 使用plt.ylim设置y坐标轴范围：(-2, 3)；
# # # 使用plt.xlabel设置x坐标轴名称：'I am x'； 使用plt.ylabel设置y坐标轴名称：'I am y'；
#
# plt.xlim((-1, 2))
# plt.ylim((-2, 3))
# plt.xlabel('I am x')
# plt.ylabel('I am y')
# # plt.show()


# 使用np.linspace定义范围以及个数：范围是(-1,2);个数是5.
# 使用print打印出新定义的范围. 使用plt.xticks设置x轴刻度：范围是(-1,2);个数是5.

new_ticks = np.linspace(-1, 2, 5)
print(new_ticks)
plt.xticks(new_ticks)

# 使用plt.yticks设置y轴刻度以及名称：刻度为[-2, -1.8, -1, 1.22, 3]；
# 对应刻度的名称为['really bad','bad','normal','good', 'really good'].
# 使用plt.show显示图像.

plt.yticks([-2, -1.8, -1, 1.22, 3], [r'$really\ bad$', r'$bad$', r'$normal$', r'$good$', r'$really\ good$'])
plt.show()
